package com.sxj.corejava.code13_JDBC;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class Test06_保存返回主键 {

    public static void main(String[] args) {
        User user = new User("alex", "123456", "13512345676", "江苏-南京");
        Integer id = insertReturnPrimaryKey(user);
        System.out.println("此时保存后返回的主键的值为:" + id);
    }

    public static Integer insertReturnPrimaryKey(User user) {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Integer id = null;
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append("   (username,password,phone,address)")
                    .append(" values ")
                    .append("   (?,?,?,?)")
                    .toString();
            // 如果保存后需要返回主键,则在获取状态集的时候需要进行设置
            // 通过参数二来设置是否返回主键
            // 其值是一个常量,有两种选择
            // RETURN_GENERATED_KEYS:返回主键,开发中一般只用返回
            // NO_GENERATED_KEYS:不返回主键
            ps = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1, user.getUsername());
            ps.setString(2, user.getPassword());
            ps.setString(3, user.getPhone());
            ps.setString(4, user.getAddress());
            // 执行DML操作
            ps.executeUpdate();
            // 当需要保存后返回主键时,其主键的值是在数据库执行完添加操作之后才会生成
            // 因此,在执行了DML操作之后,可以获取到数据库中自动生成的主键的值
            // 通过Statement对象提供的方法获取所有自动生成的键
            rs = ps.getGeneratedKeys();
            while (rs.next()) {
                // 由于自动生成的键的列名是由系统自动命名的,开发者可能不清楚其字段名
                // 如果不知道其列名是什么,可以通过第几列进行映射
                // 因为此时我们可以确定自动的生成的键有且仅有一个
                id = rs.getInt(1);
            }
        } catch (DataAccessException e) {
            throw new RuntimeException(e);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
        return id;
    }

}
